package com.haoyu.number.hashtab;

import java.util.Scanner;

/**
 * @author: 蒿雨
 * @create: 2022-01-13 18:57
 * @description:哈希表
 */
public class HashTabDemo {

    public static void main(String[] args) {
        //创建哈希表
        HashTab hashTab = new HashTab(7);
        //写一个简单得菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:id查找");
            System.out.println("exit:退出系统");
            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    //创建雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("请输入要查找得ID");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.out.println("系统退出");
                    break;
                default:
                    break;

            }
        }


    }
}

//创建一个HashTAB 管理多条链表
class HashTab {
    private EmpLinkedList[] empLinkedListArray;
    private int size;//表示共有多少条

    //构造器
    public HashTab(int size) {
        //初始化empLinkedListArray
        this.size = size;
        empLinkedListArray = new EmpLinkedList[size];
        //留一个坑,说明不要初始化每一个链表
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i] = new EmpLinkedList();
        }
    }


    //添加雇员
    public void add(Emp emp) {
        //根据员工得ID，得到该员工应当是添加到那条list
        int empLinkListNo = hashFun(emp.id);
        //将emp添加到对应得链表中
        empLinkedListArray[empLinkListNo].add(emp);

    }

    //遍历所有得链表,遍历hashtab
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i);
        }
    }

    //根据输入得id，查找该雇员
    public void findEmpById(int id) {
        //使用散列函数确定到哪条链表查找
        int empLinkedListNo = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedListNo].findEmpById(id);
        if (emp != null) {
            //找到
            System.out.printf("在第x条链表中找到该雇员=%d\n", (empLinkedListNo + 1));
        } else {
            System.out.println("没有找到该雇员");
        }
    }


    //编写散列函数，使用一个简单得取模法
    public int hashFun(int id) {
        return id % size;
    }


}

//表示一个雇员
class Emp {
    public int id;
    public String name;
    //next默认为null
    public Emp next;

    public Emp(int id, String name) {
        super();
        this.id = id;
        this.name = name;
    }
}

//创建一个EmpLinkedList，表示链表
class EmpLinkedList {
    //头指针，执行第一个Emp,因此我们这个链表得Head.是直接指向第一个Emp
    //默认null
    private Emp head;

    //添加雇员到链表
    //说明
    //1、假定，当添加雇员时，Id是自增长，即id得分配是从小到大
    //因此我们将该雇员直接加入本链表得最后即可
    public void add(Emp emp) {
        //如果是添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是第一个雇员，则使用一个辅助得指针，帮助定位到最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.next == null) {
                break;
            }
            curEmp = curEmp.next;
        }

        //退出时，直接将emp加入链表
        curEmp.next = emp;

    }

    //遍历链表得雇员信息
    public void list(int no) {
        if (head == null) {
            //说明链表为空
            System.out.println("第" + (no + 1) + "前链表为空！");
            return;
        }
        System.out.print("第" + (no + 1) + "链表得信息为空！");
        Emp curEmp = head;//辅助指针
        while (true) {
            System.out.printf("=》id=%d name = %s\t", curEmp.id, curEmp.name);
            //说明链表走完
            if (curEmp.next == null) {
                break;
            }
            curEmp = curEmp.next;
        }
        System.out.println("");
    }

    //根据id查找雇员
    //如果查找到，就返回EMP，否则就返回null
    public Emp findEmpById(int id) {
        //判断链表是否为空
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        //辅助指针
        Emp curEmp = head;
        while (true) {
            //说明找到了
            if (curEmp.id == id) {
                break;
            }
            //退出
            //说明遍历当前链表没有找到该雇员
            if (curEmp.next == null) {
                curEmp = null;
                break;
            }
            curEmp = curEmp.next;
        }

        return curEmp;

    }

}

